iT邦幫忙

2025 iThome 鐵人賽

DAY 18
0

before/after 效果的後端邏輯

在今天的文章中,我們將專注於實現 Before/After 效果 的後端邏輯。這個功能的核心是接收使用者上傳的圖片與提示詞,並透過 Google GenAI 的 API 生成新的圖片,最後將結果返回給前端。


功能需求

  1. 接收使用者上傳的圖片與提示詞。
  2. 調用 Google GenAI 的 API,生成新的圖片。
  3. 返回生成的圖片數據,供前端展示。

後端架構
後端使用 Flask 框架,並透過 Google GenAI 的 API 進行圖片生成。以下是主要的後端檔案結構:

utils/
    before_after.py  # 處理 API 請求的邏輯
    photo_mix.py     # 與 Google GenAI 的整合邏輯
    config.py        # 環境變數與設定
main.py              # Flask 主程式

  1. 環境變數設定

首先,我們需要設定 Google GenAI 的 API 金鑰,並將其存放在 .env 檔案中。

GEMINI_API_KEY=your_google_genai_api_key

config.py 中載入環境變數。

https://ithelp.ithome.com.tw/upload/images/20251002/20168630aP5MWOApeM.png

  1. 核心邏輯

以下是 before_after.py 中的主要邏輯,負責處理圖片生成請求並返回結果:

https://ithelp.ithome.com.tw/upload/images/20251002/201686304jw4TuPiHK.png

  1. API 路由實現

before_after.py 中,我們還需要實現一個 API 路由,負責接收前端的請求,並調用 BeforeAfterService 的功能:

https://ithelp.ithome.com.tw/upload/images/20251002/201686301IpF71OkSW.png

  1. 主程式整合

以下是 main.py 中處理 Before/After 效果 的主要程式碼,包含 API 路由與服務邏輯。

https://ithelp.ithome.com.tw/upload/images/20251002/201686301K84Qg2RWi.png


上一篇
DAY17
下一篇
DAY19
系列文
融合AI圖像生成技術的前端開發實戰23
圖片
  熱門推薦
圖片
{{ item.channelVendor }} | {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言